# coding=utf-8

"""
从数据库中读取脚本的内容，因为脚本内容可能会很多，
保存在Zookeeper上不适合。而数据库的查询速度是很快的。
另外选择分布式数据库已防止单点故障
"""

import pymongo

class MongoDBClient(object):
    
    def __init__(self, host='localhost', port=27017):
        self.conn = pymongo.Connection(host, port)
        self.db = self.conn.dsms
    
    
    def get_service_script(self, service):
        """
        根据服务名称，获取服务对应的脚本信息
        """
        return self.db.scripts.find({"service": service})